import type { AxiosRequestConfig } from 'axios';

// 用于存储每个请求的标识和取消函数
const pendingMap = new Map<string, AbortController>();

const getPendingUrl = (config: AxiosRequestConfig): string => {
  return [config.method, config.url].join('&');
};

export class AxiosCanceler {
  /**
   * 添加请求：将所有的请求添加到 pendingMap 中，用于后续取消在pending状态的请求
   * 调用时机：在 Axios.ts 中，拦截器在每次发送 http 请求的时候，进行拦截处理，将当前请求的信息传入改函数
   * @param config 请求配置
   */
  public addPending(config: AxiosRequestConfig): void {
    this.removePending(config);
    const url = getPendingUrl(config);
    // step1：AbortController：浏览器提供的 API，表示一个控制器对象，允许根据需要终止一个或多个 Web 请求；在请求的时候传递改实例用于终止请求
    const controller = new AbortController();
    config.signal = config.signal || controller.signal; // step:2：将实例化的controller.signal传递给config，在发起请求的时候会传递给axios
    if (!pendingMap.has(url)) {
      // step3：如果当前请求不在等待中，将其添加到等待中
      pendingMap.set(url, controller);
    }
  }

  /**
   * 清除所有等待中的请求
   */
  public removeAllPending(): void {
    pendingMap.forEach((abortController) => {
      if (abortController) {
        abortController.abort();  // step4：调用addPending时候实例化的AbortController终止请求（因为每一个AbortController实例添加到了请求参数config中，所以调用对应的abort方法即可终止）
      }
    });
    this.reset();
  }

  /**
   * 移除请求
   * @param config 请求配置
   */
  public removePending(config: AxiosRequestConfig): void {
    const url = getPendingUrl(config);
    if (pendingMap.has(url)) {
      // 如果当前请求在等待中，取消它并将其从等待中移除
      const abortController = pendingMap.get(url);
      if (abortController) {
        abortController.abort(url);
      }
      pendingMap.delete(url);
    }
  }

  /**
   * 重置
   */
  public reset(): void {
    pendingMap.clear();
  }
}
